Einsum Is All You Need: Numpy, Pytorch And Tensorflow